import fire
import os

def main():
    # 检查文件夹
    if not os.path.exists('./temp'):# 查看给定的文件/目录是否存在
        os.mkdir('./temp')# 不存在，递归创建目录
    if not os.path.exists('./output'):# 查看给定的文件/目录是否存在
        os.mkdir('./output')# 不存在，递归创建目录
    # 下载文件
    while True:
        url = input("请输入网址（输入exit退出）：")
        if 'doc88' in url:
            # 道客巴巴
            print('是道客巴巴\n')
            import doc88
            doc88.download(url)
        elif 'docin' in url:
            # 豆丁
            print('是豆丁网\n')
            import douding
            douding.download(url)
        elif 'exit' in url:
            break
        else:
            print('暂不支持')
# 直接执行脚本，语句之前和之后的代码都被执行，引用脚本，之前的语句被执行，之后的没有被执行
if __name__ == "__main__":
    fire.Fire(main)